﻿using Newtonsoft.Json;
using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using System.Threading.Tasks;
using WebApiClient.Attributes;

namespace QQRobot.Services.Models.Request
{
    /// <summary>
    /// 
    /// </summary>
    public class ImgGenerationsRequest
    {
        /// <summary>
        /// 模型
        /// </summary>
        [JsonProperty("model")]
        public string model { get; set; } = "stabilityai/stable-diffusion-3-5-large";

        /// <summary>
        /// 场景描述
        /// </summary>
        [JsonProperty("prompt")]
        public string prompt { get; set; }

        /// <summary>
        /// 场景描述
        /// </summary>
        [JsonProperty("negative_prompt")]
        public string negative_prompt { get; set; } =string.Empty;

        /// <summary>
        /// 图片尺寸
        /// </summary>
        [JsonProperty("image_size")]
        public string image_size { get; set; } = ImageSize.A;
        /// <summary>
        /// 
        /// </summary>
        [JsonProperty("batch_size")]
        public int batch_size { get; set; } = 1;

        [JsonProperty("seed")]
        public int seed { get; set;} = 1;

        [JsonProperty("num_inference_steps")]
        public int num_inference_steps { get; set; } = 25;

        [JsonProperty("guidance_scale")]
        public double guidance_scale { get; set; } = 4.5;
    }

    public class ImageSize 
    {
        public const string A = "1024x1024";
        public const string B = "512x1024";
        public const string C = "768x512";
        public const string D = "768x1024";
        public const string E = "1024x576";
        public const string F = "576x1024";
    }
}
